package com.dkd.manage.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.dkd.common.constant.DkdContants;
import com.dkd.common.exception.ServiceException;
import com.dkd.common.utils.DateUtils;
import com.dkd.manage.domain.Emp;
import com.dkd.manage.domain.Task;
import com.dkd.manage.domain.TaskDetails;
import com.dkd.manage.domain.VendingMachine;
import com.dkd.manage.domain.dto.TaskDetailsDto;
import com.dkd.manage.domain.dto.TaskDto;
import com.dkd.manage.mapper.TaskMapper;
import com.dkd.manage.service.IEmpService;
import com.dkd.manage.service.ITaskDetailsService;
import com.dkd.manage.service.ITaskService;
import com.dkd.manage.service.IVendingMachineService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.time.Duration;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 工单Service业务层处理
 *
 * @author ruoyi
 * @date 2024-08-04
 */
@Service
public class TaskServiceImpl implements ITaskService {
    @Autowired
    private TaskMapper taskMapper;

    @Autowired
    private IVendingMachineService vendingMachineService;

    @Autowired
    private IEmpService empService;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private ITaskDetailsService taskDetailsService;

    /**
     * 查询工单
     *
     * @param taskId 工单主键
     * @return 工单
     */
    @Override
    public Task selectTaskByTaskId(Long taskId) {
        return taskMapper.selectTaskByTaskId(taskId);
    }

    /**
     * 查询工单列表
     *
     * @param task 工单
     * @return 工单
     */
    @Override
    public List<Task> selectTaskList(Task task) {
        return taskMapper.selectTaskList(task);
    }

    /**
     * 新增工单
     *
     * @param task 工单
     * @return 结果
     */
    @Override
    public int insertTask(Task task) {
        task.setCreateTime(DateUtils.getNowDate());
        return taskMapper.insertTask(task);
    }

    /**
     * 修改工单
     *
     * @param task 工单
     * @return 结果
     */
    @Override
    public int updateTask(Task task) {
        task.setUpdateTime(DateUtils.getNowDate());
        return taskMapper.updateTask(task);
    }

    /**
     * 批量删除工单
     *
     * @param taskIds 需要删除的工单主键
     * @return 结果
     */
    @Override
    public int deleteTaskByTaskIds(Long[] taskIds) {
        return taskMapper.deleteTaskByTaskIds(taskIds);
    }

    /**
     * 删除工单信息
     *
     * @param taskId 工单主键
     * @return 结果
     */
    @Override
    public int deleteTaskByTaskId(Long taskId) {
        return taskMapper.deleteTaskByTaskId(taskId);
    }

    /**
     * 查询工单列表
     *
     * @param task 工单
     * @return 工单集合
     */
    @Override
    public List<Task> selectTaskVoList(Task task) {
        return taskMapper.selectTaskVoList(task);
    }


    /**
     * 新增运维, 运营工单
     *
     * @param taskDto 任务详情
     * @return 结果
     */
    @Transactional(rollbackFor = Exception.class)
    @Override
    public int insertTaskDto(TaskDto taskDto) {
        //1.查询售货机是否存在
        VendingMachine vm = vendingMachineService.selectVendingMachineByInnerCode(taskDto.getInnerCode());
        if (vm == null) {
            throw new ServiceException("售货机不存在");
        }

        // 2.校验售货机状态与工单类型是否相符
        checkCreateTaskT(vm.getVmStatus(), taskDto.getProductTypeId());
        // 3.检查设备是否有同等未完成的类型工单
        hasTask(taskDto);

        // 4.查询并校验员工是否存在
        Emp emp = empService.selectEmpById(taskDto.getUserId());
        if (emp == null) {
            throw new ServiceException("员工不存在");
        }

        // 5.校验员工区域是否匹配
        if (!emp.getRegionId().equals(vm.getRegionId())) {
            throw new ServiceException("员工与售货机区域不匹配, 无法处理此单");
        }

        // 6.将Dto对象转换po并补充属性, 保存工单
        Task task = BeanUtil.copyProperties(taskDto, Task.class);//属性复制
        task.setTaskStatus(DkdContants.TASK_STATUS_CREATE); //设置工单状态待办状态
        task.setUserName(emp.getUserName()); //执行人名称
        task.setRegionId(vm.getRegionId()); //区域id
        task.setAddr(vm.getAddr()); //地址
        task.setCreateTime(DateUtils.getNowDate()); //创建时间
        task.setTaskCode(generateTaskCode());   //工单编号

        int taskResult = taskMapper.insertTask(task);

        // 7.判断是否为补货工单
        if (taskDto.getProductTypeId().equals(DkdContants.TASK_TYPE_SUPPLY)) {
            // 8.保存工单详情
            List<TaskDetailsDto> details = taskDto.getDetails();
            if (CollUtil.isEmpty(details)) {
                throw new ServiceException("补货工单详情不能为空");
            }
            // 将dto转为po, 并补充属性
            List<TaskDetails> taskDetailsList = details.stream().map(dto -> {
                TaskDetails taskDetails = BeanUtil.copyProperties(dto, TaskDetails.class);
                taskDetails.setTaskId(task.getTaskId());
                return taskDetails;
            }).collect(Collectors.toList());
            // 批量保存工单详情
            taskDetailsService.batchInsertTaskDetails(taskDetailsList);
        }

        return taskResult;
    }

    /**
     * 取消工单
     *
     * @param task 工单
     * @return 结果
     */
    @Override
    public int cancelTask(Task task) {
        //判断工单状态是否可以取消
        // 先根据工单id查询工单
        Task taskDb = taskMapper.selectTaskByTaskId(task.getTaskId());
        if (taskDb.getTaskStatus().equals(DkdContants.TASK_STATUS_CANCEL)) {
            throw new ServiceException("工单已取消,不能再次取消");
        }
        // 判断工单状态是否已经完成, 如果已完成, 则不能取消
        if (taskDb.getTaskStatus().equals(DkdContants.TASK_STATUS_FINISH)) {
            throw new ServiceException("工单已完成,不能取消");
        }
        // 设置更新字段
        task.setUpdateTime(DateUtils.getNowDate()); //更新时间
        task.setTaskStatus(DkdContants.TASK_STATUS_CANCEL); //工单状态
        return taskMapper.updateTask(task);
    }
    /**
     * 生成并获取当天工单编号(唯一标识)
     */
    private String generateTaskCode() {
        //获取当前日期并格式化为:yyyyMMdd
        String dateStr = DateUtils.getDate().replaceAll("-", "");
        //根据当前日期生成redis锁key
        String key = "dkd.task.code." + dateStr;
        //判断key是否存在, 如果存在则获取锁, 如果不存在则创建锁
        if (!redisTemplate.hasKey(key)) {
            //如果不存在则创建锁, 并设置初始值为1,并指定过期时间为1天
            redisTemplate.opsForValue().set(key,1, Duration.ofDays(1));
            //返回生成的工单编号(日期+0001)
            return dateStr + "0001";
        }
        //如果key存在, 计数器+1 (0002), 确保字符串长度为4位
        return dateStr + StrUtil.padPre(redisTemplate.opsForValue().increment(key).toString(), 4, "0");
    }

    /**
     * 检查设备是否有同等未完成的类型工单
     *
     * @param taskDto
     */
    private void hasTask(TaskDto taskDto) {
        // 创建task条件对象, 并设置设备编号和工单类型, 以及工单状态为进行中
        Task task = new Task();
        task.setInnerCode(taskDto.getInnerCode());
        task.setProductTypeId(taskDto.getProductTypeId());
        task.setTaskStatus(DkdContants.TASK_STATUS_PROGRESS);
        // 调用Task Mapper 查询是否有未完成的工单, 如果有则抛出异常
        List<Task> taskList = taskMapper.selectTaskList(task);
        if (taskList.size() > 0) {
            throw new ServiceException("设备存在未完成的工单, 不能重复创建");
        }

    }

    /**
     * 校验售货机状态与工单类型是否相符
     *
     * @param vmStatus      售货机状态
     * @param productTypeId 工单类型
     */
    public void checkCreateTaskT(Long vmStatus, Long productTypeId) {
        //如果是投放工单，设备在运行中，抛出异常
        if (productTypeId.equals(DkdContants.TASK_TYPE_DEPLOY) && vmStatus.equals(DkdContants.VM_STATUS_RUNNING)) {
            throw new ServiceException("设备在运行中, 无法投放");
        }
        //如果是维修工单，设备不在运行中，抛出异常
        if (productTypeId.equals(DkdContants.TASK_TYPE_REPAIR) && !vmStatus.equals(DkdContants.VM_STATUS_RUNNING)) {
            throw new ServiceException("设备不在运行中, 无法维修");
        }
        //如果是补货工单，设备不在运行中，抛出异常
        if (productTypeId.equals(DkdContants.TASK_TYPE_SUPPLY) && !vmStatus.equals(DkdContants.VM_STATUS_RUNNING)) {
            throw new ServiceException("设备不在运行中, 无法补货");
        }
        //如果是撤机工单，设备不在运行中，抛出异常
        if (productTypeId.equals(DkdContants.TASK_TYPE_REVOKE) && !vmStatus.equals(DkdContants.VM_STATUS_RUNNING)) {
            throw new ServiceException("设备不在运行中, 无法撤机");
        }
    }
}
